from tensorflow import keras  # 导入 TF 子库 keras
from tensorflow.keras import layers, optimizers, datasets  # 导入 TF 子库等
import tensorflow as tf  # 导入 TF 库

if __name__ == '__main__':
    x = tf.random.normal([4, 32, 32, 3])
    layer=layers.Conv2D(16,kernel_size=3)
    out=layer(x)
    print(out.shape)
    print(layer.kernel.shape)
